import torch

# 输出当前torch的版本
print(torch.__version__)
# 当前的pytorch环境是否支持cuda
print(torch.cuda.is_available())